PyTorch CPU 版安装

要使用 Pip 安装 PyTorchCPU 版本,你可以在你的命令行中运行以下命令:

pip3 install torch torchvision torchaudio

这个命令会安装 PyTorch 以及两个与 PyTorch 相关的库:torchvisiontorchaudio。torchvision 是一个处理图像的库,它包含了许多常用的图像数据集和预训练的模型。torchaudio 是一个处理音频的库,它提供了一些音频处理的工具和数据集。

安装完成后,你可以运行以下 Python 代码来验证 PyTorch 是否安装成功:

import torch
x = torch.rand(5, 3)
print(x)

这段代码会创建一个随机初始化的张量,并打印出它的值。如果你看到了张量的值,那么说明 PyTorch 已经成功安装了。

请注意,这个命令可能需要 Python 3.8 或更高版本,以及 pip3。如果你的系统中没有安装这些软件,你可能需要先安装它们。


本文作者:Maeiee

本文链接:PyTorch CPU 版安装

版权声明:如无特别声明,本文即为原创文章,版权归 Maeiee 所有,未经允许不得转载!


喜欢我文章的朋友请随缘打赏,鼓励我创作更多更好的作品!